package com.sys.ai.workflow;

import com.agentsflex.core.llm.LlmConfig;

/**
 * @author LGY
 * Create by 2025/6/4 16:03
 */
public class CustomerLLMConfig extends LlmConfig {

    private String defaultEmbeddingModel;

    public CustomerLLMConfig() {
        this.defaultEmbeddingModel = "BAAI/bge-m3";
    }

    public CustomerLLMConfig(String model, String endpoint, String apiKey) {
        this();
        this.setEndpoint(endpoint);
        this.setApiKey(apiKey);
        this.setModel(model);
    }

    public String getDefaultEmbeddingModel() {
        return defaultEmbeddingModel;
    }
}
